#ifndef AI_FACIAL_DATA_STORE_H__
#define AI_FACIAL_DATA_STORE_H__

#include <iostream>
#include <vector>
#include "../base/messages.hpp"

namespace ai::facial_recognition {

class DataStore {
  public:
  DataStore();
  ~DataStore();

  // 保存人脸特征值
  int save_face_feature(std::string face_feature, int face_feature_length);

  // 获取已保存的所有的人脸特征值
  std::vector<FaceFeatureInfo> get_all_face_features();

  private:
  const std::string SQLITE_DB_URI = "face.db";

};

} // namespace

#endif // AI_FACIAL_DATA_STORE_H__